import time
from selenium import webdriver
from selenium.webdriver.common.by import By


class TestCase:  # 定义一个类
    def __init__(self):
        self.driver = webdriver.Chrome()  # 初始化对象属性,创建一个对象时就打开谷歌浏览器

    # 方法:搜索so,执行相关操作
    def get_so(self):
        self.driver.get('https://www.so.com/')  # 打开so搜索
        self.driver.find_element(By.ID, 'input').send_keys('python自动化测试')  # 定位到搜索框元素,输入内容
        self.driver.find_element(By.ID, 'search-button').click()  # 定位到搜索按钮,点击搜索按钮
        # 打印获取的标题
        print(self.driver.title)

        if 'python' in self.driver.title:  # 判断关键字是否在对象的标题中
            print('找到了')
        else:
            print('没找到')

    # 方法:关闭谷歌浏览器
    def quit(self):
        self.driver.quit()  # 注意这里不能使用魔术方法del,因为当对象被销毁后就不能执行quit


# 进行调试调用
if __name__ == '__main__':
    case = TestCase()  # 创建一个实例化对象
    case.get_so()  # 进行调用
    time.sleep(3)
    case.quit()
